package cn.lili.modules.verification.service.impl;

import cn.lili.cache.Cache;
import cn.lili.cache.CachePrefix;
import cn.lili.common.enums.ResultCode;
import cn.lili.common.exception.ServiceException;
import cn.lili.common.properties.VerificationCodeProperties;
import cn.lili.common.utils.StringUtils;
import cn.lili.common.vo.SerializableStream;
import cn.lili.modules.verification.SliderImageUtil;
import cn.lili.modules.verification.entity.dos.VerificationSource;
import cn.lili.modules.verification.entity.dto.VerificationDTO;
import cn.lili.modules.verification.entity.enums.VerificationEnums;
import cn.lili.modules.verification.service.VerificationService;
import cn.lili.modules.verification.service.VerificationSourceService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.io.InputStream;
import java.net.URL;
import java.util.List;
import java.util.Map;
import java.util.Random;

/**
 * 认证处理类
 *
 * @author Chopper
 * @version v1.0 2020-11-17 14:59
 */
@Slf4j
@Component
public class VerificationServiceImpl implements VerificationService {

	@Autowired
	private VerificationSourceService verificationSourceService;

	@Autowired
	private VerificationCodeProperties verificationCodeProperties;

	@Autowired
	private Cache cache;

	/**
	 * 创建校验
	 *
	 * @return 验证码参数
	 */
	@Override
	public Map<String, Object> createVerification(VerificationEnums verificationEnums, String uuid) {

		if (uuid == null) {
			throw new ServiceException(ResultCode.ILLEGAL_REQUEST_ERROR);
		}

		// 获取验证码配置
		VerificationDTO verificationDTO = verificationSourceService.getVerificationCache();

		List<VerificationSource> verificationResources = verificationDTO.getVerificationResources();
		List<VerificationSource> verificationSlider = verificationDTO.getVerificationSlider();

		Random random = new Random();
		// 随机选择需要切的图下标
		int resourceNum = random.nextInt(verificationResources.size());
		// 随机选择剪切模版下标
		int sliderNum = random.nextInt(verificationSlider.size());

		// 随机选择需要切的图片地址
		String originalResource = verificationResources.get(resourceNum).getResource();
		// 随机选择剪切模版图片地址
		String sliderResource = verificationSlider.get(sliderNum).getResource();
		// 干扰块
		String interfereResource = verificationSlider
				.get(sliderNum == verificationSlider.size() - 1 ? sliderNum - 1 : sliderNum + 1).getResource();

		try {
			// 获取缓存中的资源
			SerializableStream originalFile = getInputStream(originalResource);
			SerializableStream sliderFile = getInputStream(sliderResource);
			SerializableStream interfereSliderFile = verificationCodeProperties.getInterfereNum() > 0
					? getInputStream(interfereResource)
					: null;
			// 生成数据
			// Map<String, Object> resultMap = SliderImageUtil.pictureTemplatesCut(
			// sliderFile, interfereSliderFile, originalFile,
			// verificationCodeProperties.getWatermark(),
			// verificationCodeProperties.getInterfereNum());
			Map<String, Object> resultMap = SliderImageUtil.pictureTemplatesCut(sliderFile, interfereSliderFile,
					originalFile, null, verificationCodeProperties.getInterfereNum());
			// 生成验证参数 有效时间 默认600秒，可以自行配置
			cache.put(cacheKey(verificationEnums, uuid), resultMap.get("randomX"),
					verificationCodeProperties.getEffectiveTime());
			resultMap.put("key", cacheKey(verificationEnums, uuid));
			resultMap.put("effectiveTime", verificationCodeProperties.getEffectiveTime());
			// 移除横坐标移动距离
			resultMap.remove("randomX");
			return resultMap;
		} catch (ServiceException e) {
			throw e;
		} catch (Exception e) {
			throw new ServiceException(ResultCode.ERROR);
		}
	}

	/**
	 * 根据网络地址，获取源文件 这里简单说一下，这里是将不可序列化的inputstream序列化对象，存入redis缓存
	 *
	 * @param originalResource
	 * @return
	 */
	private SerializableStream getInputStream(String originalResource) throws Exception {

		Object object = cache.get(CachePrefix.VERIFICATION_IMAGE.getPrefix() + originalResource);
		if (object != null) {
			return (SerializableStream) object;
		}
		if (StringUtils.isNotEmpty(originalResource)) {
			URL url = new URL(originalResource);
			InputStream inputStream = url.openStream();
			SerializableStream serializableStream = new SerializableStream(inputStream);
			cache.put(CachePrefix.VERIFICATION_IMAGE.getPrefix() + originalResource, serializableStream);
			return serializableStream;
		}
		return null;
	}

	/**
	 * 预校验图片 用于前端回显
	 *
	 * @param xPos
	 *            X轴移动距离
	 * @param verificationEnums
	 *            验证key
	 * @return 验证是否成功
	 */
	@Override
	public boolean preCheck(Integer xPos, String uuid, VerificationEnums verificationEnums) {
		Integer randomX = (Integer) cache.get(cacheKey(verificationEnums, uuid));
		if (randomX == null) {
			throw new ServiceException(ResultCode.VERIFICATION_CODE_INVALID);
		}
		log.debug("{}{}", randomX, xPos);
		// 验证结果正确 && 删除标记成功
		if (Math.abs(randomX - xPos) < verificationCodeProperties.getFaultTolerant()
				&& cache.remove(cacheKey(verificationEnums, uuid))) {
			// 验证成功，则记录验证结果 验证有效时间与验证码创建有效时间一致
			cache.put(cacheResult(verificationEnums, uuid), true, verificationCodeProperties.getEffectiveTime());
			return true;
		}
		throw new ServiceException(ResultCode.VERIFICATION_ERROR);
	}

	/**
	 * 验证码校验
	 *
	 * @param uuid
	 *            用户标识
	 * @param verificationEnums
	 *            验证key
	 * @return 验证是否成功
	 */
	@Override
	public boolean check(String uuid, VerificationEnums verificationEnums) {
		// 如果有校验标记，则返回校验结果
		// if (cache.remove(cacheResult(verificationEnums, uuid))) {
		if (cacheResult(verificationEnums, uuid) != null) {
			return true;
		}
		throw new ServiceException(ResultCode.VERIFICATION_CODE_INVALID);
	}

	/**
	 * 生成缓存key 记录缓存需要验证的内容
	 *
	 * @param verificationEnums
	 *            验证码枚举
	 * @param uuid
	 *            用户uuid
	 * @return 缓存key
	 */
	public static String cacheKey(VerificationEnums verificationEnums, String uuid) {
		return CachePrefix.VERIFICATION_KEY.getPrefix() + verificationEnums.name() + uuid;
	}

	/**
	 * 生成缓存key 记录缓存验证的结果
	 *
	 * @param verificationEnums
	 *            验证码枚举
	 * @param uuid
	 *            用户uuid
	 * @return 缓存key
	 */
	public static String cacheResult(VerificationEnums verificationEnums, String uuid) {
		return CachePrefix.VERIFICATION_RESULT.getPrefix() + verificationEnums.name() + uuid;
	}

}
